//将一个给定字符串 s 根据给定的行数 numRows ，以从上往下、从左到右进行 Z 字形排列。 
//
// 比如输入字符串为 "PAYPALISHIRING" 行数为 3 时，排列如下： 
//
// 
//P   A   H   N
//A P L S I I G
//Y   I   R 
//
// 之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如："PAHNAPLSIIGYIR"。 
//
// 请你实现这个将字符串进行指定行数变换的函数： 
//
// 
//string convert(string s, int numRows); 
//
// 
//
// 示例 1： 
//
// 
//输入：s = "PAYPALISHIRING", numRows = 3
//输出："PAHNAPLSIIGYIR"
// 
//
//示例 2：
//
// 
//输入：s = "PAYPALISHIRING", numRows = 4
//输出："PINALSIGYAHRPI"
//解释：
//P     I    N
//A   L S  I G
//Y A   H R
//P     I
// 
//
// 示例 3： 
//
// 
//输入：s = "A", numRows = 1
//输出："A"
// 
//
// 
//
// 提示： 
//
// 
// 1 <= s.length <= 1000 
// s 由英文字母（小写和大写）、',' 和 '.' 组成 
// 1 <= numRows <= 1000 
// 
//
// Related Topics 字符串 👍 2461 👎 0

package com.cute.leetcode.editor.cn;

import java.util.ArrayList;
import java.util.List;

public class ZigzagConversion {
    public static void main(String[] args) {
        Solution solution = new ZigzagConversion().new Solution();
    }

    //leetcode submit region begin(Prohibit modification and deletion)
    class Solution {
        /**
         * 题解：先分多行分开保存字符串，最后拼接，仅需要一个方向一个下标
         */
        public String convert(String s, int numRows) {
            if (numRows == 1) return s;
            List<StringBuilder> list = new ArrayList<>(numRows);
            for (int i = 0; i < numRows; i++) {
                list.add(new StringBuilder());
            }
            int i = 0,flag = -1;
            for (char c : s.toCharArray()) {
                list.get(i).append(c);
                if (i == 0 || i == numRows - 1) flag = -flag; // 这里进行转向操作
                i += flag;
            }
            StringBuilder res = new StringBuilder();
            for (int j = 0; j < numRows; j++) {
                res.append(list.get(j).toString());
            }
            return res.toString();
        }

        /**
         * 暴力解法，能算出来但是效果不尽人意
         * 先求出列数，之后遍历填充即可
         */
        public String convert1(String s, int numRows) {
            int n = s.length();
            if (n<=numRows || numRows == 1 || n == 1) return s;
            int col = (n/(2*numRows-2))*(numRows-1) + ((n%(2*numRows-2)) > numRows ? (n%(2*numRows-2)-numRows) + 1 : 1);
            char[][] chars = new char[numRows][col];
            int d=0;
            int index = 0,x=0,y=0;
            while (index < n) {
                chars[x][y] = s.charAt(index++);
                if (d == 0) {
                    if (x != numRows - 1) {
                        x++;
                    }else {
                        d = 1;
                        x--;
                        y++;
                    }
                }else {
                    if (x != 0) {
                        x--;
                        y++;
                    }else {
                        d = 0;
                        x++;
                    }
                }
            }
            StringBuilder res = new StringBuilder();
            for (int i = 0; i < numRows; i++) {
                for (int j = 0; j < col; j++) {
                    if (chars[i][j] != '\u0000') {
                        res.append(chars[i][j]);
                    }
                }
            }
            return res.toString();
        }
    }
//leetcode submit region end(Prohibit modification and deletion)

}